import json
import os
from typing import Literal

from tencentcloud.common import credential
from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
from tencentcloud.common.profile.client_profile import ClientProfile
from tencentcloud.common.profile.http_profile import HttpProfile
from tencentcloud.tmt.v20180321 import models
from tencentcloud.tmt.v20180321 import tmt_client


def translate(source_text: str, from_: Literal['zh', 'en'], to_: Literal['zh', 'en']) -> str:
    try:
        # 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
        # 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
        # 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
        # secret_id = os.environ['TENCENT_SECRET_ID']
        # secret_key = os.environ['TENCENT_SECRET_KEY']
        # with open('/tmp/.secret') as f:
        #     secret_id, secret_key = map(str.strip, f.readlines())
        secret_id = 'AKIDOn9PvRqANjy7f7Z4b3JUaWKUYVM7xfCl'
        secret_key = 'SwIWjAYm0Mdov4GNWglrsncGmJsOwOwS'
        cred = credential.Credential(secret_id, secret_key)
        # 实例化一个http选项，可选的，没有特殊需求可以跳过
        httpProfile = HttpProfile()
        httpProfile.endpoint = 'tmt.tencentcloudapi.com'

        # 实例化一个client选项，可选的，没有特殊需求可以跳过
        clientProfile = ClientProfile()
        clientProfile.httpProfile = httpProfile
        # 实例化要请求产品的client对象,clientProfile是可选的
        client = tmt_client.TmtClient(cred, 'ap-shanghai', clientProfile)

        # 实例化一个请求对象,每个接口都会对应一个request对象
        req = models.TextTranslateRequest()
        params = {
            'SourceText': source_text,
            'Source': from_,
            'Target': to_,
            'ProjectId': 0,
        }
        req.from_json_string(json.dumps(params))

        # 返回的resp是一个TextTranslateResponse的实例，与请求对象对应
        resp = client.TextTranslate(req)
        # 输出json格式的字符串回包
        json_str = resp.to_json_string()
        # 响应示例: {"TargetText": "Hello", "Source": "zh", "Target": "en", "RequestId": "xxx"}
        return json.loads(json_str)['TargetText']
    except TencentCloudSDKException as err:
        return str(f'SDK异常: {err.code} - {err.message}')
    except Exception as err:
        return str(err)


def chinese_to_english(source_text):
    return translate(source_text, from_='zh', to_='en')

def english_to_chinese(source_text):
    return translate(source_text, from_='en', to_='zh')

if __name__ == '__main__':
    print(chinese_to_english('你好'))
